import time
from bedrock_agentcore.memory.controlplane import MemoryControlPlaneClient
c = MemoryControlPlaneClient()
account_id =...
region = ...

delivery_config = {
    'resources': [{
        'kinesis': {
            'dataStreamArn': f'arn:aws:kinesis:{region}:{account_id}:stream/test-memory-stream',
            'contentConfigurations': [{'type': 'MEMORY_RECORDS', 'level': 'FULL_CONTENT'}],
        }
    }]
}

memory = c.create_memory(
    name='test_stream_e2e',
    memory_execution_role_arn=f'arn:aws:iam::{account_id}:role/test-memory-stream-delivery-role',
    stream_delivery_resources=delivery_config,
)
memory_id = memory['id']
print('CREATE - id:', memory_id)
print('CREATE - streamDeliveryResources:', memory.get('streamDeliveryResources'))

print('\nWaiting for ACTIVE...')
for i in range(30):
    m = c.get_memory(memory_id)
    if m['status'] == 'ACTIVE':
        print(f'ACTIVE after {i*10}s')
        break
    time.sleep(10)

updated_config = {
    'resources': [{
        'kinesis': {
            'dataStreamArn': f'arn:aws:kinesis:{region}:{account_id}:stream/test-memory-stream',
            'contentConfigurations': [{'type': 'MEMORY_RECORDS', 'level': 'METADATA_ONLY'}],
        }
    }]
}
updated = c.update_memory(memory_id=memory_id, stream_delivery_resources=updated_config)
print('\nUPDATE - streamDeliveryResources:', updated.get('streamDeliveryResources'))

details = c.get_memory(memory_id)
print('GET    - streamDeliveryResources:', details.get('streamDeliveryResources'))
print('\nMemory', memory_id, 'left running.')